package day06;

/**
 * @author aiPlusPlus
 * @version 1.0
 * @date 2022/12/6 15:21
 */

import java.util.HashSet;
import java.util.Set;

/**
 * 有 n 个城市，其中一些彼此相连，另一些没有相连。如果城市 a 与城市 b 直接相连，且城市 b 与城市 c 直接相连，那么城市 a 与城市 c 间接相连。
 *
 * 省份 是一组直接或间接相连的城市，组内不含其他没有相连的城市。
 *
 * 给你一个 n x n 的矩阵 isConnected ，其中 isConnected[i][j] = 1 表示第 i 个城市和第 j 个城市直接相连，而 isConnected[i][j] = 0 表示二者不直接相连。
 *
 * 返回矩阵中 省份 的数量。
 *
 *
 *
 * 示例 1：
 *
 *
 * 输入：isConnected = [[1,1,0],[1,1,0],[0,0,1]]
 * 输出：2
 * 示例 2：
 *
 *
 * 输入：isConnected = [[1,0,0],[0,1,0],[0,0,1]]
 * 输出：3
 */
public class Solution4 {
    int root[];
    public int findCircleNum(int[][] isConnected) {
        int len = isConnected.length;
        root = new int[len];
        for (int i = 0; i < len; i++) {
            root[i] = i;
        }
        for (int i = 0; i < len; i++) {
            for (int j = 0; j < isConnected[i].length; j++) {
                if(isConnected[i][j]==1){
                    union(i,j);
                }
            }
        }
        Set<Integer> hs = new HashSet<>();
        for (int i = 0; i < len; i++) {
            hs.add(find(i));
        }
        return hs.size();
    }

    public int find(int x) {
        if (x == root[x]) {
            return x;
        }
        return root[x] = find(root[x]);
    }

    public void union(int x, int y) {
        int rootX = find(x);
        int rootY = find(y);
        if (rootX != rootY) {
            root[rootY] = rootX;
        }
    }
}
